from pandas import DataFrame
from entsoe.Market import EnergyPrices
from entsoe.utils import extract_records, add_timestamps

# from entsoe.config import set_config
# set_config(security_token=None, endpoint_url=None, timeout=5, retries=5,
#            retry_delay=lambda attempt: 2**attempt, max_workers=4, log_level="SUCCESS"
#           )

result = EnergyPrices(
    in_domain="10Y1001A1001A82H",  # Germany, DE-LU Biddingzone
    out_domain="10Y1001A1001A82H",
    period_start=202012312300,
    period_end=202101022300,
).query_api()

records = extract_records(result)  # Serialize <=> Convert from pydantic class to dict
records = add_timestamps(records)
df = DataFrame(records)

@split_date_range
@pagination
def query_api(params):
    results = query_and_parse(params)
    return results


@retry
def query_and_parse(params):
    responses = fetch_responses(params)
    xml_models = parse_response(responses)
    return xml_models


@unzip
def fetch_responses(params) -> list[Response]:
    response = query_core(params)
    return [response]


@handle_acknowledgement
def parse_response(response):
    _, matching_class = extract_namespace_and_find_classes(response)
    xml_model = XmlParser().from_string(response.text, matching_class)
    return xml_model

@check_if_banned
@check_service_unavailable
@rate_limit(max_calls=380, period=60)
def query_core(params):
    params_with_token = {**params, "securityToken": get_config().security_token}
    response = get(config.endpoint_url, params=params_with_token, timeout=config.timeout)
    return response

def check_service_unavailable(func):
    @wraps(func)
    def service_unavailable_wrapper(*args, **kwargs):
        response = func(*args, **kwargs)
        if response.status_code == 503:
            # @retry decorator will check for this error
            raise ServiceUnavailableError("ENTSO-E API is unavailable (HTTP 503).")
        return response

    return service_unavailable_wrapper